from dataset import get_loader_data
from encoder import Encoder
from decoder import Decoder


if __name__ == '__main__':
    encoder = Encoder()
    decoder = Decoder()

    for input, target, input_length, target_length in get_loader_data():
        # input: [batch_size, max_len]
        # target: [batch_size, max_len]

        # out: [batch_size, max_len, hidden_size]
        # hidden: [1, batch_size, hidden_size]
        out, out_length, hidden = encoder(input, input_length)
        decoder(hidden)

        # print(out.size())
        # print(out_length)
        # print(hidden.size())
        break